/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
        // 1. 先求出有多少个结点
        ListNode* cur = head;
        int cnt = 0;
        while(cur)
        {
            ++cnt;
            cur = cur->next;
        }

        // 2. 求出需要逆序多少次
        int n = cnt / k;
        
        // 3. 开始处理
        ListNode* newhead = new ListNode(0);
        ListNode* tmp = head, *next = head, *prev = newhead;   // 记录下一次要尾插的头结点
        cur = head;
        for(int i = 0; i < n; i++)  // 逆序次数
        {
            tmp = cur;  // 下一个位置尾插的头结点
            for(int j = 0; j < k; j++)  // 逆序长度
            {
                // 头插
                next = cur->next;
                cur->next = prev->next;
                prev->next = cur;
                cur = next;
            }
            prev = tmp; // 下一个位置尾插的头结点
        }

        // 翻转完之后，再把剩余的结点尾插到 prev 后面
        prev->next = cur;
        prev = newhead->next;   // 把返回结点保存一下
        delete newhead;

        return prev;
    }
};